# Uputa: staviti skriptu u direktorij sa slikama i pokrenuti.
# Rezultat: stvorena dva direktorija (test, train) sa podjeljenim slikama.
# Napomena: direktoriji test i train ne smiju postojati prije pokretanja!
import os
import shutil

clsz = {}

os.makedirs("test")
os.makedirs("train")

imgsNum = 0
files = os.listdir(".")
for f in files:
  if f[-4:] == ".nrm":
    imgsNum+=1
    if not f[0:3] in clsz.keys(): clsz[f[0:3]] = []
    clsz[f[0:3]].append(f)

print "Ucitano " + str(imgsNum) + " slika."
cur = 1
for c, imgs in clsz.iteritems():
  num = len(imgs)
  trNum = num/2
  for f in imgs[0:trNum]:
    cur+=1
    shutil.copy(f, "train/"+f)
  for f in imgs[trNum:]:
    cur+=1
    shutil.copy(f, "test/"+f)
  print "%d/%d" % (cur, imgsNum)
